#!/usr/bin/python
# author dennis
# 2022年06月15日
import socket


def write(udp_socket):

    data = input("请输入发送的信息：")
    dest_ip = input("请输入对方的IP地址：")
    dest_port = int(input("请输入对方的端口："))
    udp_socket.sendto(data.encode('utf8'), (dest_ip, dest_port))


def read(udp_socket):

    data = udp_socket.recvfrom(1024)
    print(">>>%s" % data[0].decode("utf8"))


def main():

    print("-" * 50)
    print("1.发送消息")
    print("2.接收消息")
    print("-" * 50)
    op_num = int(input("请选择功能："))

    while True:
        udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        udp_socket.bind(("", 2000))
        if op_num == 1:
            write(udp_socket)
            udp_socket.close()
            op_num = int(input("请选择功能："))
        elif op_num == 2:
            read(udp_socket)
            udp_socket.close()
            op_num = int(input("请选择功能："))
        else:
            print("输入错误，请重新输入！")
            op_num = int(input("请选择功能："))


if __name__ == '__main__':
    main()
